import java.util.Random;
import java.util.concurrent.BrokenBarrierException;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.CyclicBarrier;
import java.util.concurrent.Semaphore;
import java.util.concurrent.TimeUnit;

// JUC 辅助类 
public class JUChelper {
    public static void main(String[] args) throws InterruptedException {
        
        try {
            // CountDownLatchDemo();
            // CyclicBarrierDemo();
            SemaphoreDemo();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    // CountDownLatch
    // 减少计数，计数辅助类，用于辅助更好的利用代码，来对线程进行任务管理
    // 减少计数，是 await 线程对象等待计数结束，或者说等待所有线程完成
    public static void CountDownLatchDemo() throws InterruptedException{
        // 创建 CountDownLatch，设置初始计数
        CountDownLatch countDownLatch = new CountDownLatch(6);
        for (int i = 0; i < 6; i++) {
            new Thread(()->{
                System.out.println(Thread.currentThread().getName()+" 号线程结束");
                // 计数
                countDownLatch.countDown();
            },String.valueOf(i+1)).start();
        }
        // countdownlatch 使线程等待 countDownLatch 结束计数
        countDownLatch.await();
        System.out.println(Thread.currentThread().getName()+" 线程关闭");
    }
    // CyclicBarrier
    // 循环栅栏，同步辅助类，用于线程同步
    // 循环栅栏，是所有线程同时等待
    // 创建固定值
    private static final int NUMBER = 6;
    public static void CyclicBarrierDemo(){
        CyclicBarrier cyclicBarrier = new CyclicBarrier(NUMBER,()->{
            System.out.println("6 个线程都已完成任务，可以继续执行。");
        });
        // 完成任务过程
        for (int i = 0; i < 6; i++) {
            new Thread(()->{
                System.out.println(Thread.currentThread().getName()+" 完成任务");
                // 等待其他线程完成
                try {
                    cyclicBarrier.await();
                } catch (InterruptedException e) {
                    // TODO Auto-generated catch block
                    e.printStackTrace();
                } catch (BrokenBarrierException e) {
                    // TODO Auto-generated catch block
                    e.printStackTrace();
                }
                System.out.println(Thread.currentThread().getName()+" 收到");
            },String.valueOf(i+1)).start();
        }
    }
    // Semaphore
    // 信号量，用于获取许可，实现线程排队阻塞
    // 信号量的许可数量，允许多少个线程并行
    private static final int SNUMBER = 3;
    public static void SemaphoreDemo(){
        Semaphore s = new Semaphore(SNUMBER);
        // 六个线程等待执行任务
        for (int i = 0; i < 6; i++) {
            new Thread(()->{
                // 请求许可执行
                try {
                    s.acquire();
                    System.out.println(Thread.currentThread().getName()+" 号线程获得许可，开始执行");
                    // 模拟执行时间
                    TimeUnit.SECONDS.sleep(new Random().nextInt(5));
                    System.out.println(Thread.currentThread().getName()+" 号线程，执行完成");
                } catch (InterruptedException e) {
                    e.printStackTrace();
                } finally{
                    // 释放许可
                    s.release();
                }
            },String.valueOf(i+1)).start();
            
        }
    }
}
